from pydantic import BaseModel


class RuntimeEnv(BaseModel):
    working_dir: str
    pip: list[str] | None = None


class DeploymentSchema(BaseModel):
    name: str
    route_prefix: str
    import_path: str
    runtime_env: RuntimeEnv
    deployments: list[dict]


class DeploymentSchemaInDB(BaseModel):
    id: str | None = None
    name: str
    uuid: str
    cluster: str
    app_definition: str
